Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_identity_restore.py

Displaying Raw • Download

T8b949e# SPDX-License-Identifier: 0BSD

Tff7b72import T7ee787base64

Tff7b72import T7ee787os

Tff7b72import T7ee787shutil

Tff7b72import T7ee787tempfile

Tff7b72import T7ee787unittest

Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3MagicMockTb4b4b4, Te6edf3patch

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787identity_manager Tff7b72import Te6edf3IdentityManager

Tff7b72class T56d364TestIdentityRestoreTb4b4b4(Te6edf3unittestTff7b72.Td2a8ffTestCaseTb4b4b4)Tb4b4b4:

Tff7b72def Td2a8ffsetUpTb4b4b4(Tff7b72selfTb4b4b4)Tb4b4b4:

Tff7b72selfTff7b72.Td2a8fftemp_dir Tff7b72= Te6edf3tempfileTff7b72.Td2a8ffmkdtempTb4b4b4(Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffidentity_manager Tff7b72= Te6edf3IdentityManagerTb4b4b4(Tff7b72selfTff7b72.Td2a8fftemp_dirTb4b4b4)

Tff7b72def Td2a8fftearDownTb4b4b4(Tff7b72selfTb4b4b4)Tb4b4b4:

Te6edf3shutilTff7b72.Td2a8ffrmtreeTb4b4b4(Tff7b72selfTff7b72.Td2a8fftemp_dirTb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.IdentityTa5d6ff"Tb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_manager.DatabaseProviderTa5d6ff"Tb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_manager.DatabaseSchemaTa5d6ff"Tb4b4b4)

Tff7b72def Td2a8fftest_restore_identity_from_bytesTb4b4b4(

Tff7b72selfTb4b4b4,

Te6edf3mock_schemaTb4b4b4,

Te6edf3mock_providerTb4b4b4,

Te6edf3mock_rns_identityTb4b4b4,

Tb4b4b4)Tb4b4b4:

T8b949e# Setup mock identity

Te6edf3mock_id_instance Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_id_instanceTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6fftest_hash_32_bytes_long_01234567Ta5d6ff"

Te6edf3mock_id_instanceTff7b72.Td2a8ffget_private_keyTff7b72.Td2a8ffreturn_value Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6fftest_private_keyTa5d6ff"

Te6edf3mock_rns_identityTff7b72.Td2a8fffrom_bytesTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3mock_id_instance

Te6edf3identity_bytes Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffsome_identity_bytesTa5d6ff"

Te6edf3result Tff7b72= Tff7b72selfTff7b72.Td2a8ffidentity_managerTff7b72.Td2a8ffrestore_identity_from_bytesTb4b4b4(Te6edf3identity_bytesTb4b4b4)

Te6edf3identity_hash Tff7b72= Te6edf3mock_id_instanceTff7b72.Td2a8ffhashTff7b72.Td2a8ffhexTb4b4b4(Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertEqualTb4b4b4(Te6edf3resultTb4b4b4[Ta5d6ff"Ta5d6ffhashTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3identity_hashTb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertEqualTb4b4b4(Te6edf3resultTb4b4b4[Ta5d6ff"Ta5d6ffdisplay_nameTa5d6ff"Tb4b4b4]Tb4b4b4, Ta5d6ff"Ta5d6ffRestored IdentityTa5d6ff"Tb4b4b4)

T8b949e# Verify files were created

Te6edf3identity_dir Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Tff7b72selfTff7b72.Td2a8fftemp_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentitiesTa5d6ff"Tb4b4b4, Te6edf3identity_hashTb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertTrueTb4b4b4(Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffexistsTb4b4b4(Te6edf3identity_dirTb4b4b4)Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertTrueTb4b4b4(Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffexistsTb4b4b4(Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3identity_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentityTa5d6ff"Tb4b4b4)Tb4b4b4)Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertTrueTb4b4b4(Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffexistsTb4b4b4(Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3identity_dirTb4b4b4, Ta5d6ff"Ta5d6ffmetadata.jsonTa5d6ff"Tb4b4b4)Tb4b4b4)Tb4b4b4)

T8b949e# Verify private key was written

Tff7b72with Tffa657openTb4b4b4(Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3identity_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentityTa5d6ff"Tb4b4b4)Tb4b4b4, Ta5d6ff"Ta5d6ffrbTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:

Tff7b72selfTff7b72.Td2a8ffassertEqualTb4b4b4(Te6edf3fTff7b72.Td2a8ffreadTb4b4b4(Tb4b4b4)Tb4b4b4, Ta5d6ffbTa5d6ff"Ta5d6fftest_private_keyTa5d6ff"Tb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.IdentityTa5d6ff"Tb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_manager.DatabaseProviderTa5d6ff"Tb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_manager.DatabaseSchemaTa5d6ff"Tb4b4b4)

Tff7b72def Td2a8fftest_restore_identity_from_base32Tb4b4b4(

Tff7b72selfTb4b4b4,

Te6edf3mock_schemaTb4b4b4,

Te6edf3mock_providerTb4b4b4,

Te6edf3mock_rns_identityTb4b4b4,

Tb4b4b4)Tb4b4b4:

T8b949e# Setup mock identity

Te6edf3mock_id_instance Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_id_instanceTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6fftest_hash_32_bytes_long_01234567Ta5d6ff"

Te6edf3mock_id_instanceTff7b72.Td2a8ffget_private_keyTff7b72.Td2a8ffreturn_value Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6fftest_private_keyTa5d6ff"

Te6edf3mock_rns_identityTff7b72.Td2a8fffrom_bytesTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3mock_id_instance

Te6edf3identity_bytes Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffsome_identity_bytesTa5d6ff"

Te6edf3base32_value Tff7b72= Te6edf3base64Tff7b72.Td2a8ffb32encodeTb4b4b4(Te6edf3identity_bytesTb4b4b4)Tff7b72.Td2a8ffdecodeTb4b4b4(Ta5d6ff"Ta5d6ffutf-8Ta5d6ff"Tb4b4b4)

Te6edf3result Tff7b72= Tff7b72selfTff7b72.Td2a8ffidentity_managerTff7b72.Td2a8ffrestore_identity_from_base32Tb4b4b4(Te6edf3base32_valueTb4b4b4)

Te6edf3identity_hash Tff7b72= Te6edf3mock_id_instanceTff7b72.Td2a8ffhashTff7b72.Td2a8ffhexTb4b4b4(Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertEqualTb4b4b4(Te6edf3resultTb4b4b4[Ta5d6ff"Ta5d6ffhashTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3identity_hashTb4b4b4)

T8b949e# Verify from_bytes was called with the decoded bytes

Te6edf3mock_rns_identityTff7b72.Td2a8fffrom_bytesTff7b72.Td2a8ffassert_called_withTb4b4b4(Te6edf3identity_bytesTb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.IdentityTa5d6ff"Tb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_manager.DatabaseProviderTa5d6ff"Tb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_manager.DatabaseSchemaTa5d6ff"Tb4b4b4)

Tff7b72def Td2a8fftest_restore_identity_from_base32_with_display_nameTb4b4b4(

Tff7b72selfTb4b4b4,

Te6edf3mock_schemaTb4b4b4,

Te6edf3mock_providerTb4b4b4,

Te6edf3mock_rns_identityTb4b4b4,

Tb4b4b4)Tb4b4b4:

Te6edf3mock_id_instance Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_id_instanceTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6fftest_hash_32_bytes_long_01234567Ta5d6ff"

Te6edf3mock_id_instanceTff7b72.Td2a8ffget_private_keyTff7b72.Td2a8ffreturn_value Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6fftest_private_keyTa5d6ff"

Te6edf3mock_rns_identityTff7b72.Td2a8fffrom_bytesTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3mock_id_instance

Te6edf3identity_bytes Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffsome_identity_bytesTa5d6ff"

Te6edf3base32_value Tff7b72= Te6edf3base64Tff7b72.Td2a8ffb32encodeTb4b4b4(Te6edf3identity_bytesTb4b4b4)Tff7b72.Td2a8ffdecodeTb4b4b4(Ta5d6ff"Ta5d6ffutf-8Ta5d6ff"Tb4b4b4)

Te6edf3result Tff7b72= Tff7b72selfTff7b72.Td2a8ffidentity_managerTff7b72.Td2a8ffrestore_identity_from_base32Tb4b4b4(

Te6edf3base32_valueTb4b4b4,

Te6edf3display_nameTff7b72=Ta5d6ff"Ta5d6ffImported NameTa5d6ff"Tb4b4b4,

Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertEqualTb4b4b4(Te6edf3resultTb4b4b4[Ta5d6ff"Ta5d6ffdisplay_nameTa5d6ff"Tb4b4b4]Tb4b4b4, Ta5d6ff"Ta5d6ffImported NameTa5d6ff"Tb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.IdentityTa5d6ff"Tb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_manager.DatabaseProviderTa5d6ff"Tb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_manager.DatabaseSchemaTa5d6ff"Tb4b4b4)

Tff7b72def Td2a8fftest_restore_identity_from_base32_blank_display_name_uses_defaultTb4b4b4(

Tff7b72selfTb4b4b4,

Te6edf3mock_schemaTb4b4b4,

Te6edf3mock_providerTb4b4b4,

Te6edf3mock_rns_identityTb4b4b4,

Tb4b4b4)Tb4b4b4:

Te6edf3mock_id_instance Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_id_instanceTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6fftest_hash_32_bytes_long_01234567Ta5d6ff"

Te6edf3mock_id_instanceTff7b72.Td2a8ffget_private_keyTff7b72.Td2a8ffreturn_value Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6fftest_private_keyTa5d6ff"

Te6edf3mock_rns_identityTff7b72.Td2a8fffrom_bytesTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3mock_id_instance

Te6edf3identity_bytes Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffsome_identity_bytesTa5d6ff"

Te6edf3base32_value Tff7b72= Te6edf3base64Tff7b72.Td2a8ffb32encodeTb4b4b4(Te6edf3identity_bytesTb4b4b4)Tff7b72.Td2a8ffdecodeTb4b4b4(Ta5d6ff"Ta5d6ffutf-8Ta5d6ff"Tb4b4b4)

Te6edf3result Tff7b72= Tff7b72selfTff7b72.Td2a8ffidentity_managerTff7b72.Td2a8ffrestore_identity_from_base32Tb4b4b4(

Te6edf3base32_valueTb4b4b4,

Te6edf3display_nameTff7b72=Ta5d6ff"Ta5d6ff Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertEqualTb4b4b4(Te6edf3resultTb4b4b4[Ta5d6ff"Ta5d6ffdisplay_nameTa5d6ff"Tb4b4b4]Tb4b4b4, Ta5d6ff"Ta5d6ffRestored IdentityTa5d6ff"Tb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.IdentityTa5d6ff"Tb4b4b4)

Tff7b72def Td2a8fftest_restore_identity_invalid_bytesTb4b4b4(Tff7b72selfTb4b4b4, Te6edf3mock_rns_identityTb4b4b4)Tb4b4b4:

Te6edf3mock_rns_identityTff7b72.Td2a8fffrom_bytesTff7b72.Td2a8ffreturn_value Tff7b72= Tff7b72None

Tff7b72with Tff7b72selfTff7b72.Td2a8ffassertRaisesTb4b4b4(Tf85149ValueErrorTb4b4b4) Tff7b72as Te6edf3cmTb4b4b4:

Tff7b72selfTff7b72.Td2a8ffidentity_managerTff7b72.Td2a8ffrestore_identity_from_bytesTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6ffinvalidTa5d6ff"Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertInTb4b4b4(Ta5d6ff"Ta5d6ffCould not load identity from bytesTa5d6ff"Tb4b4b4, Tffa657strTb4b4b4(Te6edf3cmTff7b72.Td2a8ffexceptionTb4b4b4)Tb4b4b4)

Tf0883e@patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.IdentityTa5d6ff"Tb4b4b4)

Tff7b72def Td2a8fftest_restore_identity_invalid_base32Tb4b4b4(Tff7b72selfTb4b4b4, Te6edf3mock_rns_identityTb4b4b4)Tb4b4b4:

Tff7b72with Tff7b72selfTff7b72.Td2a8ffassertRaisesTb4b4b4(Tf85149ValueErrorTb4b4b4) Tff7b72as Te6edf3cmTb4b4b4:

Tff7b72selfTff7b72.Td2a8ffidentity_managerTff7b72.Td2a8ffrestore_identity_from_base32Tb4b4b4(Ta5d6ff"Ta5d6ffinvalid-base32-!!!Ta5d6ff"Tb4b4b4)

Tff7b72selfTff7b72.Td2a8ffassertInTb4b4b4(Ta5d6ff"Ta5d6ffInvalid base32 identityTa5d6ff"Tb4b4b4, Tffa657strTb4b4b4(Te6edf3cmTff7b72.Td2a8ffexceptionTb4b4b4)Tb4b4b4)

Tff7b72if Tff7b72__name__ Tff7b72== Ta5d6ff"Ta5d6ff__main__Ta5d6ff"Tb4b4b4:

Te6edf3unittestTff7b72.Td2a8ffmainTb4b4b4(Tb4b4b4)